Class VGG16Encoder

Inheritance Relationships

Base Type

Class Documentation

encoder_decoder_model.vgg_encoder.VGG16Encoder : public encoder_decoder_model.cnn_basenet.CNNBaseModel

Public Functions

__init__(self, phase)

Implement a feature encoding class based on VGG16.

Parameters

phase

encode(self, input_tensor, name)

Encode the input tensor according to the VGG16 framework.

Parameters
  • input_tensor

  • name

  • flags

Returns

Output VGG16 encoding feature